/*
 * Copyright (c) Microsoft Corporation
 *
 * All rights reserved.
 *
 * MIT License
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated
 * documentation files (the "Software"), to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and
 * to permit persons to whom the Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all copies or substantial portions of
 * the Software.
 *
 * THE SOFTWARE IS PROVIDED *AS IS*, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO
 * THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
 * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

package com.microsoft.azuretools.authmanage;

import com.microsoft.azuretools.authmanage.models.AuthMethodDetails;
import com.microsoft.azuretools.sdkmanage.AccessTokenAzureManager;
import com.microsoft.azuretools.sdkmanage.AzureManager;
import com.microsoft.azuretools.sdkmanage.ServicePrincipalAzureManager;

public enum AuthMethod {
    AD(new AccessTokenAzureManager.AccessTokenAzureManagerFactory(AdAuthManager.getInstance())),
    DC(new AccessTokenAzureManager.AccessTokenAzureManagerFactory(DCAuthManager.getInstance())),
    SP(new ServicePrincipalAzureManager.ServicePrincipalAzureManagerFactory()),
    ;

    private final AzureManagerFactory azureManagerFactory;

    AuthMethod(final AzureManagerFactory azureManagerFactory) {
        this.azureManagerFactory = azureManagerFactory;
    }

    public AzureManager createAzureManager(final AuthMethodDetails authMethodDetails) {
        return this.azureManagerFactory.factory(authMethodDetails);
    }

    public AuthMethodDetails restoreAuth(final AuthMethodDetails authMethodDetails) {
        return this.azureManagerFactory.restore(authMethodDetails);
    }

    public BaseADAuthManager getAdAuthManager() {
        if (azureManagerFactory instanceof AdAuthManagerBuilder) {
            return ((AdAuthManagerBuilder) azureManagerFactory).getInstance();
        }

        throw new IllegalArgumentException("No AD Auth manager instance for authentication method " + this);
    }
}
